6.4--Bellman-Ford的队列优化

//6.4Bellman-Ford的队列优化

#include<stdio.h>
// N的值比n大1，M的值比m大1 
#define N 6 
#define M 8
int main()
{
    freopen("data.in","r",stdin);
    int n,m,i,j,k;
    int u[M],v[M],w[M];
    int first[N],next[M];//first记录各点的第一条边，next递归存下下一条； 
    int dis[N]={0},book[N]={0}; //记录距离和队列中已有的点 
    int que[101]={0},head=1,tail=1;
    int inf=999999;
    scanf("%d%d",&n,&m);
    //初始化 
    for(i=1;i<=n;i++)
        dis[i]=inf;
    dis[1]=0;
    for(i=1;i<=n;i++)
        first[i]=-1;
    //读入边
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d",&u[i],&v[i],&w[i]);
        //建立邻接表的关键
        next[i]=first[u[i]];
        first[u[i]]=i; 
    }
    //1号顶点入队
    que[tail]=1;
    tail++;
    book[1]=1;
    while(head<tail)
    {
        k=first[que[head]];
        while(k!=-1)
        {
            if(dis[v[k]]>dis[u[k]]+w[k])
            {
                dis[v[k]]=dis[u[k]]+w[k];
                //book数组用来判断顶点v[k]是否在队列中 
                if(book[v[k]]==0)
                {
                    que[tail]=v[k];
                    tail++;
                    book[v[k]]=1;
                }
            }
            k=next[k];
        }
        //出队 
        book[que[head]]=0;
        head++; 
    }   
    for(i=1;i<=n;i++)
            printf("%d ",dis[i]);
    return 0;   
}